tensor([[[0.4900, 0.4900],
         [0.4900, 0.4900]],

        [[0.2500, 0.2500],
         [0.2500, 0.2500]],

        [[0.2600, 0.2600],
         [0.2600, 0.2600]]], grad_fn=<RoundBackward1>)